from tempfile import TemporaryFile
from tempfile import NamedTemporaryFile

# 需要在程序执行时创建一个临时文件或目录，并希望使用完之后可以自动销毁
def demo_1():
    # 临时文件，文本模式w+t, 二进制模式使用w+b
    with TemporaryFile('w+t') as f:
        f.write('Hello World\n')
        f.write('Test kepler\n')
        f.seek(0)
        data = f.read()
        print(data)

# 输出临时文件名, delete=False设定文件关闭时不会自动删除
def demo_2():
    with NamedTemporaryFile('w+t', delete=False) as f:
        print('filename is :', f.name)

# 自定义临时文件
def demo_3():
    f = NamedTemporaryFile(prefix='testtempfile', suffix='.txt', dir='/tmp')
    print(f.name)

if __name__ == '__main__':
    demo_1()
